from admin_chioce_p.admin_choice import book_import, salesreport
from user_chioce_p.user_choice import browse_book, payment
from user_p.user import reg, login, admin_login


def start_menu():

    tip = '''
***************** 欢迎来到Woniu_book ********************
***************** 请选择操作菜单 ************************
** 1. 用户注册  2. 用户登录  3. 管理员登录  4. 退出系统 **
    '''
    print(tip)
    option = input("请输入你的操作选项：")
    if option == "1":
        reg()
    elif option == "2":
        login()
    elif option=="3":
        admin_login()
    elif option=="4":
        exit('谢谢使用')
    else:
        print("输入错误请重新输入")
        start_menu()


# 构建用户登录后的操作菜单
def user_menu():
    tip = '''
********************* 恭喜登录蜗牛图书馆 *****************************
*********************** 请选择操作菜单 *******************************
************ 1. 浏览图书+加购物车   2. 结账   3. 退出系统    *********
        '''
    print(tip)
    option = input("请输入你的操作选项：")
    if option == "1":
        browse_book()
    elif option == "2":
        payment()
    elif option == "3":
        exit('谢谢使用')
    else:
        print("输入错误请重新输入")
        user_menu()

# 构建管理员登录后的操作菜单
def admin_menu():
    tip = '''
********************* 恭喜登录蜗牛图书馆 *****************************
*********************** 请选择操作菜单 *******************************
************ 1. 图书入库   2. 浏览销售报表   3. 退出系统    ***********
        '''
    print(tip)
    option = input("请输入你的操作选项：")
    if option == "1":
        book_import()
    elif option == "2":
        salesreport()
    elif option == "3":
        exit('谢谢使用')
    else:
        print("输入错误请重新输入")
        admin_menu()

if __name__ == '__main__':
    start_menu()